'use strict';

module.exports = app => {
  const mongoose = app.mongoose;
  const Schema = mongoose.Schema;

  const NotificationSchema = new Schema({
    title: { type: String },
    avatar: { type: String },
    startTime: { type: Date },
    endTime: { type: Date },
    content: { type: String },
  });

  return mongoose.model('Notification', NotificationSchema);
};
